package com.heima.behavior.service.impl;

import com.alibaba.fastjson.JSON;
import com.heima.behavior.dto.LikesBehaviorDto;
import com.heima.behavior.dto.UpdateArticleMessage;
import com.heima.behavior.entity.ApLikesBehavior;
import com.heima.behavior.service.ILikeService;
import com.heima.common.dto.ResponseResult;
import com.heima.common.dto.User;
import com.heima.common.util.AppThreadLocalUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.stereotype.Service;

import java.util.Date;

@Service
public class LikeServiceImpl implements ILikeService {

    @Autowired
    MongoTemplate mongoTemplate; // mongo数据库

    @Autowired
    KafkaTemplate<String, String> kafkaTemplate; // kafka 消息中间件, 发送消息
    @Value("${topic.like_behavior}")
    String LikeTopic; // 发送消息到 哪个话题
    /**
     * 点赞功能
     * @param dto
     * @return
     */
    @Override
    public ResponseResult saveLike(LikesBehaviorDto dto) {
        //1.构建用户点赞行为对象
        ApLikesBehavior behavior = new ApLikesBehavior();
        //2. 用线程取用户信息
        User user = AppThreadLocalUtil.get();
        //3. 取到说明是用户登录, 取不到说明是游客登录, 用设备id
        if (user != null){
            // 写入用户id
            behavior.setUserId(user.getUserId());
        } else {
            // 写入设备id, 从前端出入
            behavior.setEquipmentId(dto.getEquipmentId());
        }
        // 写入文章、动态、评论等ID, 从前端传入
        behavior.setArticleId(dto.getArticleId());
        // 写入点赞内容类型, 0文章, 1动态
        behavior.setType(0);
        // 写入喜欢操作方式: 0 点赞   1 取消点赞, 从前端传入
        behavior.setOperation(dto.getOperation());
        // 写入创建时间
        behavior.setCreatedTime(new Date());

        //4. 保存用户的点赞记录
        // 判断当前用户对当前文章是否已经有点赞记录, 构建查询条件
        Query query = new Query();
        // 文章、动态、评论等ID条件
        query.addCriteria(Criteria.where("articleId").is(dto.getArticleId()));
        // 判断是不是用户登录进行点赞, 不是用户, 就设备id,          是的话, 就用户id
        if (user == null){
            // 非用户登录, 设备id 条件
            query.addCriteria(Criteria.where("equipmentId").is(dto.getEquipmentId()));
        } else {
            // 用户登录, 用户id 条件
            query.addCriteria(Criteria.where("userId").is(user.getUserId()));
        }
        // 调用mongo进行条件查询
        ApLikesBehavior one = mongoTemplate.findOne(query, ApLikesBehavior.class);
        // 如果没有查到记录, 说明没点赞, 要啊进行新增
        if (one == null){
            // 如果喜欢操作方式 == 0 说明是0 点赞 进行保存
            if (dto.getOperation() == 0){
                mongoTemplate.save(behavior);
                // 发送消息到kafka
                sendMessage(dto);
            }
        } else {
            // 如果查到了记录, 判断当前的操作和数据库中的 喜欢操作方式 是否一致
            if (one.getOperation() != dto.getOperation()){
                // 如果不一致,修改操作类型,更新记录
                one.setOperation(dto.getOperation());
                // mongo进行修改
                mongoTemplate.save(one);
                // 发送消息到kafka
                sendMessage(dto);
            }
        }
        // 点赞功能完成, 返回成功标志
        return ResponseResult.okResult();
    }

    /**
     * 点赞记录的查询
     * @param dto
     * @return
     */
    @Override
    public ResponseResult getLikes(LikesBehaviorDto dto) {
        //1. 判断当前用户对当前文章是否已经有点赞记录, 构建查询条件
        Query query = new Query();
        //2. 文章、动态、评论等ID条件
        query.addCriteria(Criteria.where("articleId").is(dto.getArticleId()));
        //3. 判断是不是用户登录进行点赞, 不是用户, 就设备id,          是的话, 就用户id
        if (dto.getUserId() == null){
            // 非用户登录, 设备id 条件
            query.addCriteria(Criteria.where("equipmentId").is(dto.getEquipmentId()));
        } else {
            // 用户登录, 用户id 条件
            query.addCriteria(Criteria.where("userId").is(dto.getUserId()));
        }
        //4. 调用mongo进行条件查询
        ApLikesBehavior one = mongoTemplate.findOne(query, ApLikesBehavior.class);
        //5. 返回查询结果到前端
        return ResponseResult.okResult(one);
    }


    /**
     * 发送消息到kafka
     * @param dto
     */
    private void sendMessage(LikesBehaviorDto dto){
        //1. 发送点赞消息到kafka
        // 构建消息对象
        UpdateArticleMessage message = new UpdateArticleMessage();
        // 写入操作类型 0 阅读 1 点赞 2 评论 3 收藏
        message.setType(1);
        // 写入文章ID, 从前端传入
        message.setArticleId(dto.getArticleId());
        // 写入修改数据的增量，可为正负
        message.setAdd(1);
        //2. 如果前端出来的 喜欢的操作方式 == 1, 说明是取消点赞
        if (dto.getOperation() == 1){
            // 写入修改数据的增量 为-1
            message.setAdd(-1);
        }
        //3. 定义value  转json串
        String value = JSON.toJSONString(message);
        //4. 调用kafka 发送消息
        kafkaTemplate.send(LikeTopic, value);
    }

}
